import os
import requests
import json

API_URL = "https://api.gitcode.com/api/v5/chat/completions"
headers = {
    "Authorization": f"Bearer Gj-gxvr8_SCec3vEZzzAiAyF",
}

def query(payload):
    response = requests.post(API_URL, headers=headers, json=payload, stream=True)
    for line in response.iter_lines():
        if not line.startswith(b"data:"):
            continue
        if line.strip() == b"data:[DONE]":
            return
        yield json.loads(line.decode("utf-8").lstrip("data:").rstrip("/n"))

chunks = query({
    "messages": [
        {
            "role": "user",
            "content": "告诉我一个有关宇宙的有趣事实？"
        }
    ],
    "model": "MoonshotAI/Kimi-K2-Thinking",
    "stream":  True,
    "max_tokens": 4092,
    "temperature": 0.6,
    "top_p": 0.7,
    "top_k": 50,
    "min_p": 0,
    "frequency_penalty": 0,
    "thinking_budget": 4096
})

for chunk in chunks:
    print(chunk["choices"][0]["delta"]["content"], end="")